672a8990090608772c05c92c34d9505e99f3e73d,cassandra-scheduler/src/test/java/com/mesosphere/dcos/cassandra/scheduler/CassandraSchedulerTest.java,CassandraSchedulerTest,testRegistered,#,187

Before Change


    @Test
    public void testRegistered() throws Exception {
        scheduler.registered(driver, frameworkId, masterInfo);
        final Protos.FrameworkID frameworkID = stateStore.fetchFrameworkId();
        assertEquals(frameworkID, this.frameworkId);
        final Phase currentPhase = stageManager.getCurrentPhase();
        assertTrue(currentPhase instanceof ReconciliationPhase);

After Change


    @Test
    public void testRegistered() throws Exception {
        scheduler.registered(driver, frameworkId, masterInfo);
        final Protos.FrameworkID frameworkID = stateStore.fetchFrameworkId().get();
        assertEquals(frameworkID, this.frameworkId);
        final Phase currentPhase = planManager.getCurrentPhase().get();
        assertTrue(currentPhase instanceof ReconciliationPhase);